home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / pprint.pyo (.txt) < prev    next >
Python Compiled Bytecode  |  2005-10-18  |  9KB  |  347 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyo (Python 2.4)
  3.  
  4. """Support to pretty-print lists, tuples, & dictionaries recursively.
  5.  
  6. Very simple, but useful, especially in debugging data structures.
  7.  
  8. Classes
  9. -------
  10.  
  11. PrettyPrinter()
  12.     Handle pretty-printing operations onto a stream using a configured
  13.     set of formatting parameters.
  14.  
  15. Functions
  16. ---------
  17.  
  18. pformat()
  19.     Format a Python object into a pretty-printed representation.
  20.  
  21. pprint()
  22.     Pretty-print a Python object to a stream [default is sys.stdout].
  23.  
  24. saferepr()
  25.     Generate a 'standard' repr()-like value, but protect against recursive
  26.     data structures.
  27.  
  28. """
  29. import sys as _sys
  30. from cStringIO import StringIO as _StringIO
  31. __all__ = [
  32.     'pprint',
  33.     'pformat',
  34.     'isreadable',
  35.     'isrecursive',
  36.     'saferepr',
  37.     'PrettyPrinter']
  38. _commajoin = ', '.join
  39. _id = id
  40. _len = len
  41. _type = type
  42.  
  43. def pprint(object, stream = None, indent = 1, width = 80, depth = None):
  44.     '''Pretty-print a Python object to a stream [default is sys.stdout].'''
  45.     printer = PrettyPrinter(stream = stream, indent = indent, width = width, depth = depth)
  46.     printer.pprint(object)
  47.  
  48.  
  49. def pformat(object, indent = 1, width = 80, depth = None):
  50.     '''Format a Python object into a pretty-printed representation.'''
  51.     return PrettyPrinter(indent = indent, width = width, depth = depth).pformat(object)
  52.  
  53.  
  54. def saferepr(object):
  55.     '''Version of repr() which can handle recursive data structures.'''
  56.     return _safe_repr(object, { }, None, 0)[0]
  57.  
  58.  
  59. def isreadable(object):
  60.     '''Determine if saferepr(object) is readable by eval().'''
  61.     return _safe_repr(object, { }, None, 0)[1]
  62.  
  63.  
  64. def isrecursive(object):
  65.     '''Determine if object requires a recursive representation.'''
  66.     return _safe_repr(object, { }, None, 0)[2]
  67.  
  68.  
  69. class PrettyPrinter:
  70.     
  71.     def __init__(self, indent = 1, width = 80, depth = None, stream = None):
  72.         '''Handle pretty printing operations onto a stream using a set of
  73.         configured parameters.
  74.  
  75.         indent
  76.             Number of spaces to indent for each level of nesting.
  77.  
  78.         width
  79.             Attempted maximum number of columns in the output.
  80.  
  81.         depth
  82.             The maximum depth to print out nested structures.
  83.  
  84.         stream
  85.             The desired output stream.  If omitted (or false), the standard
  86.             output stream available at construction will be used.
  87.  
  88.         '''
  89.         indent = int(indent)
  90.         width = int(width)
  91.         self._depth = depth
  92.         self._indent_per_level = indent
  93.         self._width = width
  94.         if stream is not None:
  95.             self._stream = stream
  96.         else:
  97.             self._stream = _sys.stdout
  98.  
  99.     
  100.     def pprint(self, object):
  101.         self._stream.write(self.pformat(object) + '\n')
  102.  
  103.     
  104.     def pformat(self, object):
  105.         sio = _StringIO()
  106.         self._format(object, sio, 0, 0, { }, 0)
  107.         return sio.getvalue()
  108.  
  109.     
  110.     def isrecursive(self, object):
  111.         return self.format(object, { }, 0, 0)[2]
  112.  
  113.     
  114.     def isreadable(self, object):
  115.         (s, readable, recursive) = self.format(object, { }, 0, 0)
  116.         if readable:
  117.             pass
  118.         return not recursive
  119.  
  120.     
  121.     def _format(self, object, stream, indent, allowance, context, level):
  122.         level = level + 1
  123.         objid = _id(object)
  124.         if objid in context:
  125.             stream.write(_recursion(object))
  126.             self._recursive = True
  127.             self._readable = False
  128.             return None
  129.         
  130.         rep = self._repr(object, context, level - 1)
  131.         typ = _type(object)
  132.         sepLines = _len(rep) > self._width - 1 - indent - allowance
  133.         write = stream.write
  134.         if sepLines:
  135.             r = getattr(typ, '__repr__', None)
  136.             if issubclass(typ, dict) and r is dict.__repr__:
  137.                 write('{')
  138.                 if self._indent_per_level > 1:
  139.                     write((self._indent_per_level - 1) * ' ')
  140.                 
  141.                 length = _len(object)
  142.                 if length:
  143.                     context[objid] = 1
  144.                     indent = indent + self._indent_per_level
  145.                     items = object.items()
  146.                     items.sort()
  147.                     (key, ent) = items[0]
  148.                     rep = self._repr(key, context, level)
  149.                     write(rep)
  150.                     write(': ')
  151.                     self._format(ent, stream, indent + _len(rep) + 2, allowance + 1, context, level)
  152.                     if length > 1:
  153.                         for key, ent in items[1:]:
  154.                             rep = self._repr(key, context, level)
  155.                             write(',\n%s%s: ' % (' ' * indent, rep))
  156.                             self._format(ent, stream, indent + _len(rep) + 2, allowance + 1, context, level)
  157.                         
  158.                     
  159.                     indent = indent - self._indent_per_level
  160.                     del context[objid]
  161.                 
  162.                 write('}')
  163.                 return None
  164.             
  165.             if (issubclass(typ, list) or r is list.__repr__ or issubclass(typ, tuple)) and r is tuple.__repr__:
  166.                 if issubclass(typ, list):
  167.                     write('[')
  168.                     endchar = ']'
  169.                 else:
  170.                     write('(')
  171.                     endchar = ')'
  172.                 if self._indent_per_level > 1:
  173.                     write((self._indent_per_level - 1) * ' ')
  174.                 
  175.                 length = _len(object)
  176.                 if length:
  177.                     context[objid] = 1
  178.                     indent = indent + self._indent_per_level
  179.                     self._format(object[0], stream, indent, allowance + 1, context, level)
  180.                     if length > 1:
  181.                         for ent in object[1:]:
  182.                             write(',\n' + ' ' * indent)
  183.                             self._format(ent, stream, indent, allowance + 1, context, level)
  184.                         
  185.                     
  186.                     indent = indent - self._indent_per_level
  187.                     del context[objid]
  188.                 
  189.                 if issubclass(typ, tuple) and length == 1:
  190.                     write(',')
  191.                 
  192.                 write(endchar)
  193.                 return None
  194.             
  195.         
  196.         write(rep)
  197.  
  198.     
  199.     def _repr(self, object, context, level):
  200.         (repr, readable, recursive) = self.format(object, context.copy(), self._depth, level)
  201.         if not readable:
  202.             self._readable = False
  203.         
  204.         if recursive:
  205.             self._recursive = True
  206.         
  207.         return repr
  208.  
  209.     
  210.     def format(self, object, context, maxlevels, level):
  211.         """Format object for a specific context, returning a string
  212.         and flags indicating whether the representation is 'readable'
  213.         and whether the object represents a recursive construct.
  214.         """
  215.         return _safe_repr(object, context, maxlevels, level)
  216.  
  217.  
  218.  
  219. def _safe_repr(object, context, maxlevels, level):
  220.     typ = _type(object)
  221.     if typ is str:
  222.         if 'locale' not in _sys.modules:
  223.             return (repr(object), True, False)
  224.         
  225.         if "'" in object and '"' not in object:
  226.             closure = '"'
  227.             quotes = {
  228.                 '"': '\\"' }
  229.         else:
  230.             closure = "'"
  231.             quotes = {
  232.                 "'": "\\'" }
  233.         qget = quotes.get
  234.         sio = _StringIO()
  235.         write = sio.write
  236.         for char in object:
  237.             if char.isalpha():
  238.                 write(char)
  239.                 continue
  240.             write(qget(char, repr(char)[1:-1]))
  241.         
  242.         return ('%s%s%s' % (closure, sio.getvalue(), closure), True, False)
  243.     
  244.     r = getattr(typ, '__repr__', None)
  245.     if issubclass(typ, dict) and r is dict.__repr__:
  246.         if not object:
  247.             return ('{}', True, False)
  248.         
  249.         objid = _id(object)
  250.         if maxlevels and level > maxlevels:
  251.             return ('{...}', False, objid in context)
  252.         
  253.         if objid in context:
  254.             return (_recursion(object), False, True)
  255.         
  256.         context[objid] = 1
  257.         readable = True
  258.         recursive = False
  259.         components = []
  260.         append = components.append
  261.         level += 1
  262.         saferepr = _safe_repr
  263.         for k, v in object.iteritems():
  264.             (krepr, kreadable, krecur) = saferepr(k, context, maxlevels, level)
  265.             (vrepr, vreadable, vrecur) = saferepr(v, context, maxlevels, level)
  266.             append('%s: %s' % (krepr, vrepr))
  267.             if readable and kreadable:
  268.                 pass
  269.             readable = vreadable
  270.             if krecur or vrecur:
  271.                 recursive = True
  272.                 continue
  273.         
  274.         del context[objid]
  275.         return ('{%s}' % _commajoin(components), readable, recursive)
  276.     
  277.     if (issubclass(typ, list) or r is list.__repr__ or issubclass(typ, tuple)) and r is tuple.__repr__:
  278.         if issubclass(typ, list):
  279.             if not object:
  280.                 return ('[]', True, False)
  281.             
  282.             format = '[%s]'
  283.         elif _len(object) == 1:
  284.             format = '(%s,)'
  285.         elif not object:
  286.             return ('()', True, False)
  287.         
  288.         format = '(%s)'
  289.         objid = _id(object)
  290.         if maxlevels and level > maxlevels:
  291.             return (format % '...', False, objid in context)
  292.         
  293.         if objid in context:
  294.             return (_recursion(object), False, True)
  295.         
  296.         context[objid] = 1
  297.         readable = True
  298.         recursive = False
  299.         components = []
  300.         append = components.append
  301.         level += 1
  302.         for o in object:
  303.             (orepr, oreadable, orecur) = _safe_repr(o, context, maxlevels, level)
  304.             append(orepr)
  305.             if not oreadable:
  306.                 readable = False
  307.             
  308.             if orecur:
  309.                 recursive = True
  310.                 continue
  311.         
  312.         del context[objid]
  313.         return (format % _commajoin(components), readable, recursive)
  314.     
  315.     rep = repr(object)
  316.     if rep:
  317.         pass
  318.     return (rep, not rep.startswith('<'), False)
  319.  
  320.  
  321. def _recursion(object):
  322.     return '<Recursion on %s with id=%s>' % (_type(object).__name__, _id(object))
  323.  
  324.  
  325. def _perfcheck(object = None):
  326.     import time as time
  327.     if object is None:
  328.         object = [
  329.             ('string', (1, 2), [
  330.                 3,
  331.                 4], {
  332.                 5: 6,
  333.                 7: 8 })] * 100000
  334.     
  335.     p = PrettyPrinter()
  336.     t1 = time.time()
  337.     _safe_repr(object, { }, None, 0)
  338.     t2 = time.time()
  339.     p.pformat(object)
  340.     t3 = time.time()
  341.     print '_safe_repr:', t2 - t1
  342.     print 'pformat:', t3 - t2
  343.  
  344. if __name__ == '__main__':
  345.     _perfcheck()
  346.  
  347.